package cn.mw.cmdb.config;

import lombok.extern.slf4j.Slf4j;
import org.eclipse.paho.client.mqttv3.*;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;

@Slf4j
@Configuration
public class MQTTProviderConfig {

    @Value("${spring.mqtt.username}")
    private String username;

    @Value("${spring.mqtt.password}")
    private String password;

    @Value("${spring.mqtt.url}")
    private String hostUrl;

    @Value("${spring.mqtt.client}")
    private String clientId;


    /**
     * 客户端对象
     */
    private MqttClient client;

    @PostConstruct
    public void init() {
//        connect();
    }

    public void connect() {
        try {
            String clientIdKey = clientId + System.currentTimeMillis();
            //创建MQTT客户端对象
            client = new MqttClient(hostUrl, clientIdKey);
            //连接设置
            MqttConnectOptions options = new MqttConnectOptions();
            //是否清空session，设置false表示服务器会保留客户端的连接记录（订阅主题，qos）,客户端重连之后能获取到服务器在客户端断开连接期间推送的消息
            //设置为true表示每次连接服务器都是以新的身份
            options.setCleanSession(true);
            //设置连接用户名
            options.setUserName(username);
            //设置连接密码
            options.setPassword(password.toCharArray());
            //设置超时时间，单位为秒
            options.setConnectionTimeout(100);
            //设置心跳时间 单位为秒，表示服务器每隔 1.5*20秒的时间向客户端发送心跳判断客户端是否在线
            options.setKeepAliveInterval(20);
            //设置遗嘱消息的话题，若客户端和服务器之间的连接意外断开，服务器将发布客户端的遗嘱信息
            options.setWill("willTopic", (clientId + "与服务器断开连接").getBytes(), 0, false);
            client.setCallback(new MQTTPushCallback());
            client.connect(options);
            subScribeMQTT();
        } catch (MqttException e) {
            e.printStackTrace();
        }

    }

    /**
     * 发布消息
     *
     * @param topic
     * @param messageInfo
     */
    public void publish(String topic, String messageInfo) {
        MqttMessage mqttMessage = new MqttMessage();
        mqttMessage.setQos(1);
        mqttMessage.setRetained(true);
        MqttTopic mqttTopic = client.getTopic(topic);
        MqttDeliveryToken token;
        try {
            //将指定消息发布到主题，但不等待消息传递完成，返回的token可用于跟踪消息的传递状态
            //一旦此方法干净地返回，消息就已被客户端接受发布，当连接可用，将在后台完成消息传递。
            String message = messageInfo;
            mqttMessage.setPayload(message.getBytes());
            token = mqttTopic.publish(mqttMessage);
            token.waitForCompletion();
        } catch (MqttException e) {
            e.printStackTrace();
        }
    }

    public void subScribeMQTT(){
        String[] topic = {"V6800Upload/#", "V5008Upload/#"};
        int[] Qos = {1, 1};
        subScribe(topic, Qos);
    }


    /**
     * 订阅主题
     *
     * @param topic
     * @param Qos
     */
    public void subScribe(String[] topic, int[] Qos) {
        try {
            if (topic != null) {
                log.info("订阅了::" + String.join("；", topic));
                client.subscribe(topic, Qos);
            }
        } catch (MqttException e) {
            e.printStackTrace();
        }
    }

}
